package com.zhonghui.controller;

import com.zhonghui.common.core.result.FwResult;
import com.zhonghui.model.Category;
import com.zhonghui.service.CategoryService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

@RestController
@RequestMapping("/category")
public class CategoryController {

    @Autowired
    private CategoryService categoryService;

    @GetMapping
    public FwResult get(){

        List<Category> list =
                categoryService.lambdaQuery().eq(Category::getParentId, 0).list();

        list.forEach(n -> {
            n.setTreeList(categoryService.lambdaQuery().eq(Category::getParentId, n.getId()).list());
        });

        return FwResult.ok(list);
    }

}
